from locust import HttpUser, task, between


class XiaomiUser(HttpUser):
    # 模拟用户的等待时间（每个任务之间的间隔时间）
    wait_time = between(1, 3)

    @task(1)  # 任务权重为1
    def test_homepage(self):
        # 测试商城首页
        self.client.get("/")

    @task(2)  # 任务权重为2
    def test_login(self):
        # 测试登录接口
        self.client.post("/user/login", json={"username": "wwz", "password": "123"})
